#pragma once

#include "iostream"
#include "vector"
#include "algorithm"

using namespace std;
/*HJJ QQ479287006
 *给定一个整数数组prices，其中第  prices[i] 表示第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

 

示例 1:

输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
示例 2:

输入: prices = [1]
输出: 0

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * */

int maxProfit(vector<int>& prices) {

    vector<vector<int>> dp(prices.size(),vector<int>(4,0));
    dp[0][0] -= prices[0];//需要初始化

    for (int i = 1; i <prices.size(); ++i) {
        //买入 是买⼊股票状态并不是说今天已经就买⼊股票，⽽是说保存买⼊   股票的状态即：可能是前⼏天买⼊的，之后⼀直没操作，所以保持买⼊股票的状态
        dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3], dp[i - 1][1]) - prices[i]);
        //保持卖出股票状态 注意这里没有 今天卖出的  、//如果比上有今天卖出的 似乎就没有冷冻期了
        dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
        //今天卖出了股票
        dp[i][2] = dp[i - 1][0] + prices[i];
        //冷冻=前一个卖出的
        dp[i][3] = dp[i - 1][2];
    }

    //返回 冷冻期的值和卖出的最大
    return max(dp[prices.size() - 1][3],max(dp[prices.size() - 1][1], dp[prices.size() - 1][2]));

}


//public int maxProfit(int[] prices) {
//    if(prices == null || prices.length == 0){
//        return 0;
//    }
//    //状态转移图：
//    //       持股               不持股
//    //     ↙-----、   卖出    ↙-----、
//    //    持股-----↑--------→不持股---↑
//    //      |
//    //      |卖
//    //      |出
//    //      ↓
//    //    冷冻期(期间什么都不能干，冷冻期结束后转为不持股状态)
//    //
//    int[][] dp = new int[prices.length][3];
//    //dp[i][x]第i天进入(处于)x状态（0.不持股(不代表今天刚好不持股)，1.持股()，2.冷冻期）
//    //不持股
//    dp[0][0] = 0;
//    //持股
//    dp[0][1] = -prices[0];
//    //冷冻期
//    dp[0][2] = 0;
//    for(int i = 1;i < prices.length;i++){
//        //第i天不持股可以从两种状态转移而来，1.第i-1天不持股，今天仍不买股票，保持不持股状态。2.冷冻期结束了，但是今天不买股票。
//        dp[i][0] = Math.max(dp[i-1][0], dp[i-1][2]);
//        //第i天持股可从两种状态转移而来，1.第i-1天不持股(包含昨天是冷冻期，冷冻期结束后转为不持股状态和昨天本身就不持股这两种情况)，今天买股票。2.第i-1天持股，今天不卖出，保持持股状态。
//        dp[i][1] = Math.max(dp[i - 1][0] - prices[i], dp[i - 1][1]);
//        //只有第i - 1天卖出了股票，第i天才处于冷冻期。
//        dp[i][2] = dp[i-1][1] + prices[i];
//    }
//    //只有最后一天不持股（不持股状态）或者前一天已经卖掉了（今天为冷冻期）这两种情况手里是拿着钱的，最大值在二者中产生。
//    return Math.max(dp[prices.length - 1][0], dp[prices.length - 1][2]);
//}


int maxProfit(vector<int>& prices) {

    vector<vector<int>> dp(prices.size(),vector<int>(4,0));
    dp[0][0] -= prices[0];//需要初始化


    //（0.不持股，1.持股，2.冷冻期）

    //dp[i][0]=dp[i-1][0] ,dp[i-1][1]+price[i] ,dp[i-1][2]
    dp[0][0] -= prices[0];//需要初始化
    for (int i = 1; i <prices.size(); ++i) {
        //买入
        dp[i][0]=max(dp[i-1][0], max(dp[i-1][1]+price[i],dp[i-1][2]));

        dp[i][1]= max(max( dp[i-1][1],dp[i-1][0]-prices[i]),dp[i-1][2]-prices[i]);

        dp[i][2]=dp[i-1][1]+prices[i];



    }

    //返回 冷冻期的值和卖出的最大
    return max(dp[prices.size() - 1][0], dp[prices.size()  - 1][2]);

}
